# 导包
import unittest
from unittest1 import TestDemo1, TestDemo2


# 创建测试套件
suite = unittest.TestSuite()

# 将测试用例加载到测试套件中
# TODO 写法一:类名("方法名")
# suite.addTest(TestDemo1("test_method1"))
# suite.addTest(TestDemo2('test_b'))

# TODO 写法二:addTests, tests可以是元组和列表
tests = (
    TestDemo2('test_b'),
    TestDemo1('test_method2'),
)
# suite.addTests(tests)

# TODO 添加测试类中所有方法, unittest.makeSuite(类名)
suite.addTest(unittest.makeSuite(TestDemo1))

# 执行测试用例，实例化TextTestRunner类
runner = unittest.TextTestRunner(verbosity=2)

# 使用run()方法运行测试套件（即运行测试套件中的所有用例）
runner.run(suite)

